import os
import pandas as pd

# 设置包含数据文件的文件夹路径
folder_path = 'data/year'

# 获取该文件夹中所有的文件名
files = [f for f in os.listdir(folder_path) if f.endswith('.csv')]

# 创建一个空的 DataFrame 用于存储所有的数据
all_data = pd.DataFrame()

# 遍历所有 CSV 文件，将它们读取并合并到 all_data 中
for file in files:
    file_path = os.path.join(folder_path, file)
    # 读取每个 CSV 文件，并将数据添加到 all_data
    data = pd.read_csv(file_path)
    all_data = pd.concat([all_data, data], ignore_index=True)

# 将合并后的数据写入新的 CSV 文件
all_data.to_csv('merged_data.csv', index=False)

print("数据合并完成，保存为 'merged_data.csv'")
